import torch
import timm
import torchvision.models as models
from torchsummary import summary


# model = timm.create_model('densenet169')
# print(model)
model = models.alexnet(num_classes=10)
summary(model, input_size=(3, 32, 32), device='cpu')